package array.others;

import java.util.HashMap;
import java.util.Map;

/**
 * 给你一个整数数组 arr ，数组中的每个整数 互不相同 。另有一个由整数数组构成的数组 pieces，其中的整数也 互不相同 。请你以 任意顺序 连接 pieces 中的数组以形成 arr 。但是，不允许 对每个数组 pieces[i] 中的整数重新排序。
 * <p>
 * 如果可以连接 pieces 中的数组形成 arr ，返回 true ；否则，返回 false 。
 * <p>
 * 来源：力扣（LeetCode）
 * <p>
 * 链接：<a href="https://leetcode.cn/problems/check-array-formation-through-concatenation">1640. 能否连接形成数组</a>
 */
public class N1640_CheckArrayFormationThroughConcatenation {
    public boolean canFormArray(int[] arr, int[][] pieces) {
        int n = arr.length, m = pieces.length;
        Map<Integer, Integer> index = new HashMap<>();
        for (int i = 0; i < m; i++)
            index.put(pieces[i][0], i);

        for (int i = 0; i < n;) {
            if (!index.containsKey(arr[i]))
                return false;
            int j = index.get(arr[i]), len = pieces[j].length;
            for (int k = 0; k < len; k++) {
                if (arr[i + k] != pieces[j][k]) return false;
            }
            i = i + len;
        }
        return true;
    }
}
